import matplotlib.pyplot as plt

# 计算斐波那契数列的第 n 项
def fibonacci(n):
    dp = [0] * (n + 1)  # 初始化数组
    if n > 0:
        dp[1] = 1

    for i in range(2, n + 1):
        dp[i] = dp[i - 1] + dp[i - 2]

    return dp[n], dp  # 返回结果和完整序列

# 绘制斐波那契数列的图像并保存
def plot_fibonacci_sequence(n):
    _, sequence = fibonacci(n)
    plt.plot(range(n + 1), sequence, marker='o')
    plt.title(f"斐波那契数列前 {n} 项")
    plt.xlabel("n")
    plt.ylabel("F(n)")
    plt.grid(True)
    filename = "fibonacci_sequence.png"
    plt.savefig(filename)  # 保存图像到本地
    print(f"图形已保存为 {filename}")
    plt.show()

# 输入并计算
n = int(input("请输入一个非负整数 n: "))
result, _ = fibonacci(n)
print(f"F({n}) = {result}")

plot_fibonacci_sequence(n)  # 绘制并保存图像